#pragma once

#include <iostream>
#include <signal.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>

const char *rooot = "/";
const char *dev_null = "/dev/null";

void Daemon(bool ischroot, bool isclosefd)
{
    // 1.忽略可能会引起程序异常退出信号
    signal(SIGCHLD, SIG_IGN); // 忽略子进程退出信号
    signal(SIGHUP, SIG_IGN);  // 忽略挂起信号

    // 2.不要让自己成为组长
    if (fork() > 0)
        exit(0); // 父进程退出

    // 3.设置让自己成为一个新的会话，后面的代码子进程执行
    setsid();

    // 4.每一个进程都有自己的工作目录CWD，是否将当前进程CWD更改为根目录
    if (ischroot)
        chdir(rooot);

    // 5.已经把变成守护进程，不需要使用用户输入、输出、错误进行关联了

    if (isclosefd)
    {
        close(0);
        close(1);
        close(2);
    }
    else
    {
        int fd = open(dev_null, O_RDWR);
        if (fd > 0)
        {
            dup2(fd, 0);
            dup2(fd, 1);
            dup2(fd, 2);
            close(fd);
        }
    }
}